from deep_translator import GoogleTranslator

def translate_text(text, source_lang='auto', target_lang='zh-CN'):
    """
    翻译文本，支持中英互译
    
    参数:
        text (str): 要翻译的文本
        source_lang (str): 源语言代码，默认为auto自动检测
        target_lang (str): 目标语言代码，默认为zh-CN(简体中文)
    
    返回:
        str: 翻译后的文本
    """
    return GoogleTranslator(source=source_lang, target=target_lang).translate(text)

def test():
    # 中译英示例
    chinese_text = "你好，世界"
    en_translation = translate_text(chinese_text, source_lang='zh-CN', target_lang='en')
    print(f"中文: {chinese_text}")
    print(f"英文翻译: {en_translation}")

    # 英译中示例
    english_text = "keep it up, you are awesome"
    zh_translation = translate_text(english_text, source_lang='en', target_lang='zh-CN')
    print(f"英文: {english_text}")
    print(f"中文翻译: {zh_translation}")

# 仅在直接运行此文件时执行测试
if __name__ == "__main__":
    test()